#!/usr/bin/env bash

TRITON_BUILD=/root/triton/build/cmake.linux-x86_64-cpython-3.13

set -e -x

TRITON_TENSOR_LAYOUT=`find ${TRITON_BUILD} -name triton-tensor-layout`

BLOCKED_ENCODING="#ttg.blocked<{sizePerThread = [4], threadsPerWarp = [32], warpsPerCTA = [4], order = [0] }>"

TENSOR_TYPE="tensor<32xf32>"

${TRITON_TENSOR_LAYOUT} -l "${BLOCKED_ENCODING}" -t "${TENSOR_TYPE}" > 4.txt
